# dispatcher/model_dispatcher.py
from app.services.doubao.model_registry import MODEL_REGISTRY


class ModelDispatcher:
    @staticmethod
    def get_model(task_type: str, **kwargs):
        if task_type not in MODEL_REGISTRY:
            raise ValueError(f"未知任务类型: {task_type}")
        return MODEL_REGISTRY[task_type](**kwargs)
